/*
 *    Copyright (c) 2018-2025, huangpengfei All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: huangpengfei (641655770@qq.com)
 */

package com.alpha.eceasy.mall.util;

import cn.hutool.core.util.StrUtil;
import com.alpha.eceasy.auth.LoginUser;
import com.alpha.eceasy.mall.api.entity.MallCustomer;
import lombok.experimental.UtilityClass;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.context.SecurityContextHolder;

import java.util.Collection;
import java.util.List;

/**
 * 安全工具类
 *
 * @author L.cm
 */
@UtilityClass
public class SecurityUtils {

    /**
     * 获取Authentication
     */
    public Authentication getAuthentication() {
        return SecurityContextHolder.getContext().getAuthentication();
    }

    /**
     * 获取用户
     *
     * @param authentication authentication
     * @return LoginUser
     * <p>
     */
    @SuppressWarnings("unchecked")
    public LoginUser<MallCustomer> getLoginUser(Authentication authentication) {
        Object principal = authentication.getPrincipal();
        return principal instanceof LoginUser ? (LoginUser<MallCustomer>) principal : null;
    }

    /**
     * 获取登录用户
     */
    public LoginUser<MallCustomer> getLoginUser() {
        Authentication authentication = getAuthentication();
        return getLoginUser(authentication);
    }

    /**
     * 获取用户详情
     */
    public MallCustomer getUserInfo() {
        return getLoginUser().getUserInfo();
    }

    /**
     * 获取用户guid
     */
    public String getUserGuid() {
        return getLoginUser().getUserInfo().getGuid();
    }

    /**
     * 获取用户角色信息
     *
     * @return 角色集合
     */
    public List<String> getRoles() {
        Authentication authentication = getAuthentication();
        Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities();

        return authorities.stream()
                .filter(granted -> StrUtil.startWith(granted.getAuthority(), "ROLE_"))
                .map(granted -> StrUtil.removePrefix(granted.getAuthority(), "ROLE_"))
                .toList();
    }
}
